#!/bin/bash

#SBATCH -p npu
#SBATCH --mem=64g
#SBATCH --gres=npu:1
#SBATCH -c 12
#SBATCH -t 7-00:00:00
#SBATCH --output=exp_020.out

#需要将数据集路径修改为实际路径

# source activate mlfold-test
export CPU_AFFINITY_CONF=1
export TASK_QUEUE_ENABLE=2

python ./training.py \
           --path_for_outputs "./exp_020" \
           --path_for_training_data "/dataset/pdb_2021aug02" \
           --num_examples_per_epoch 1000 \
           --save_model_every_n_epochs 50
